matplotlib animation tip